import { pinyin } from 'pinyin-pro'

function py (str) {
  return pinyin(str, { toneType: 'none', type: 'array' }).join('')
}

function py2 (str) {
  return pinyin(str, { toneType: 'none', type: 'array' }).join('_').toUpperCase()
}

export default function install (Vue) {
  Vue.prototype.$pinyin = function (str) {
    return py(str)
  }
  Vue.prototype.$pinyin_uppercase = function (str) {
    return py2(str)
  }
}
